package com.plusjun.gulimall.order.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

@Configuration
public class MyRabbitConfig {
    @Resource
    RabbitTemplate rabbitTemplate;
    /**
     * 配置使用Json的消息转换器
     *
     * @return
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制ReabbitTemplate--发送端确认：
     * @PostConstruct: 在所在类的对象创建完之后执行
     *  1：服务器收到消息就回调
     *      1）、配置yaml：spring.rabbitmqpublisher-confirm-type=CORRELATED
     *      2）、设置确认回调
     *  2：消息正确抵达队列进行回调
     *     1）、yaml：spring.rabbitmq.publisher-returns=true
     *               spring.rabbitmq.template.mandatory=true
     *     2）设置回调
     *
     * 定制ReabbitTemplate--消费端确认：(保证每个消息被正确消费，此时才可以broker删除这个消息）。
     *  1、默认是自动确认的，只要消息接收到，客户端会自动确认，服务端就会移除这个消息
     *      问题：我们收到很多消息，自动回复给服务器ACK，只有一个消息处理成功就宕机了，然后剩下的消息都丢失了
     *      解决：开启消费者手动确认。只要我们没有明确告诉MQ，消息被签收，没有ACK，就一直时unacked,即使服务器宕机，
     *             消息也不会都是，会重新变为Ready，下一次新的Consumer连接就发给他
     *  2、签收消息：
     *      Long deliveryTag = message.getMessageProperties().getDeliveryTag()；
     *      deliveryTag在channel中的消息序号（自增的）
     *      在消费消息后签收： channel.basicAck(deliveryTag,false);//false:非批量模式
     *      拒收： channel.basicNack(deliveryTag,false,false);
     */
    @PostConstruct
    public void initRabbitTemplate() {
        // 设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 只要消息抵达服务器就执行回调
             * @param correlationData：当前消息的唯一关联数据（消息的唯一id）
             * @param ack： 消息是否成功收到 (只要消息抵达 Broker就为true）
             * @param cause： 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                // 修改消息状态

                System.out.println("confirm..." +correlationData + "===" + ack + "===" + cause);
            }
        });

        // 设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就触发这个失败回调
             * @param message 投递失败的消息详细信息
             * @param replyCode 回复的状态码
             * @param replyText 回复的文本内容
             * @param exchange 当时这个消息发送给哪个交换机
             * @param routingKey 当时这个消息用哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                System.out.println("message["+message+"]==replyCode[" + replyCode +
                        "] == replyText["+replyText+ "] == exchange["+ exchange + "] == routingKey[" + routingKey +"]");
            }
        });
    }
}
